from YoloDataCreator import YOLO_Dataset_Creator

img_path = r'.\example\origin\img'
label_path = r'.\example\origin\json'
save_path = r'.\example\yolo'

creator = YOLO_Dataset_Creator(img_path=img_path, label_path=label_path, save_path=save_path)

# 创建yolo数据集
creator.dataset_main()

# 可视化检查
visfile = r'.\example\origin\vis'
creator.visual_json_main(visfile=visfile)